simplify code using an inverse dictionary in python